diff --git a/tests/mocks/database/db/driver.php b/tests/mocks/database/db/driver.php
index 2cf54b9..86f7efd 100644
--- a/tests/mocks/database/db/driver.php
+++ b/tests/mocks/database/db/driver.php
@@ -37,4 +37,4 @@
 
 }
 
-class CI_DB extends Mock_Database_DB_QueryBuilder {}
\ No newline at end of file
+class CI_DB extends CI_DB_query_builder {}
\ No newline at end of file
diff --git a/tests/mocks/database/db/querybuilder.php b/tests/mocks/database/db/querybuilder.php
deleted file mode 100644
index 3f22526..0000000
--- a/tests/mocks/database/db/querybuilder.php
+++ /dev/null
@@ -1,3 +0,0 @@
-<?php
-
-class Mock_Database_DB_QueryBuilder extends CI_DB_query_builder {}
\ No newline at end of file